package com.nowcoder.community.event;

import com.alibaba.fastjson.JSONObject;
import com.nowcoder.community.entity.DiscussPost;
import com.nowcoder.community.entity.Event;
import com.nowcoder.community.entity.Message;
import com.nowcoder.community.service.DiscussPostService;
import com.nowcoder.community.service.ElasticsearchService;
import com.nowcoder.community.service.MessageService;
import com.nowcoder.community.util.CommunityConstant;
import com.nowcoder.community.util.GenerateUtil;
import com.qiniu.common.QiniuException;
import com.qiniu.common.Zone;
import com.qiniu.http.Response;
import com.qiniu.storage.Configuration;
import com.qiniu.storage.UploadManager;
import com.qiniu.util.Auth;
import com.qiniu.util.StringMap;
import org.apache.kafka.clients.consumer.ConsumerRecord;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.kafka.annotation.KafkaListener;
import org.springframework.scheduling.concurrent.ThreadPoolTaskScheduler;
import org.springframework.stereotype.Component;

import java.io.File;
import java.io.IOException;
import java.util.Date;
import java.util.HashMap;
import java.util.Map;
import java.util.concurrent.Future;
import java.util.concurrent.ScheduledFuture;

/**
 * @BelongsProject: community
 * @BelongsPackage: com.nowcoder.community.event
 * @Author: Master_XXJS
 * @CreateTime: 2023-01-07  15:20
 * @Description: TODO
 * @Version: 1.0
 */
@Component
public class EventConsumer implements CommunityConstant {

    private static final Logger logger = LoggerFactory.getLogger(EventConsumer.class);

    @Autowired
    private MessageService messageService;
    @Autowired
    private DiscussPostService discussPostService;
    @Autowired
    private ElasticsearchService elasticsearchService;
    @Autowired
    private ThreadPoolTaskScheduler taskScheduler;

    @Value("${wk.image.command}")
    private String wkImageCommand;
    @Value("${wk.image.storage}")
    private String wkImageStorage;
    @Value("${qiniu.key.access}")
    private String accessKey;
    @Value("${qiniu.key.secret}")
    private String secretKey;
    @Value("${qiniu.bucket.share.name}")
    private String shareBucketName;

    /**
     * 获取消费者的信息
     * 被动触发，不需要主动的调用，只需要自动调用生产者就行
     * @param record 事件对象
     */
    @KafkaListener(topics = {TOPIC_COMMENT,TOPIC_FOLLOW,TOPIC_LIKE})
    public void handleEvent(ConsumerRecord record){
        if (record == null || record.value() == null){
            logger.error("消息的内容为空！");
            return;
        }

        //将 json 数据的消息，转换为 事件event对象
        Event event = JSONObject.parseObject(record.value().toString(), Event.class);
        if (event == null){
            logger.error("消息的格式错误！");
            return;
        }

        //发送站内通知，数据库中 存放事件对象信息的就是 message 表（存放所有的消息）
        Message message = new Message();
        message.setFromId(SYSTEM_USER_ID);
        message.setToId(event.getEntityUserId());
        message.setConversationId(event.getTopic());
        message.setCreateTime(new Date());

        //消息的内容。我们需要添加多一点东西，因为页面需要进行显示下面的三个参数
        Map<String,Object> content = new HashMap<>();
        content.put("userId",event.getUserId());
        content.put("entityType",event.getEntityType());
        content.put("entityId",event.getEntityId());
        //判断监听获取到的 事件对象 的是否有数据，有的话，就将里面的数据取出放到我们新的 容器
        if(!event.getData().isEmpty()){
            for (Map.Entry<String,Object> entry : event.getData().entrySet()){
                content.put(entry.getKey(),entry.getValue());
            }
        }
        //将 content 消息的内容存入 message 对象中
        message.setContent(JSONObject.toJSONString(content));

        //存入数据库中
        messageService.addMessage(message);
    }

    // 消费发帖事件
    @KafkaListener(topics = {TOPIC_PUBLISH})
    public void handlePublishMessage(ConsumerRecord record) {
        if (record == null || record.value() == null) {
            logger.error("消息的内容为空!");
            return;
        }

        //将 数据 转换为 事件event对象
        Event event = JSONObject.parseObject(record.value().toString(), Event.class);
        if (event == null) {
            logger.error("消息格式错误!");
            return;
        }


        DiscussPost post = discussPostService.selectByDiscussPostId(event.getEntityId());
        //发帖，或则评论之后，将帖子重新查询出来，更新到  es 中
        elasticsearchService.saveDiscussPost(post);
    }

    // 删除帖事件
    @KafkaListener(topics = {TOPIC_DELETE})
    public void handleDeleteMessage(ConsumerRecord record) {
        if (record == null || record.value() == null) {
            logger.error("消息的内容为空!");
            return;
        }

        //将 数据 转换为 事件event对象
        Event event = JSONObject.parseObject(record.value().toString(), Event.class);
        if (event == null) {
            logger.error("消息格式错误!");
            return;
        }

        //将 es 中的帖子删除
        elasticsearchService.deleteDiscussPost(event.getEntityId());
    }

    //分享帖子的事件,生成图片、PDF
    @KafkaListener(topics = {TOPIC_SHARE})
    public void handleShareMessage(ConsumerRecord record){
        if (record == null || record.value() == null) {
            logger.error("消息的内容为空!");
            return;
        }

        //将 数据 转换为 事件event对象
        Event event = JSONObject.parseObject(record.value().toString(), Event.class);
        if (event == null) {
            logger.error("消息格式错误!");
            return;
        }

        //先进行转换，因为直接获取的出来的都是  Object 对象，我们需要转换
        String htmlUrl = (String) event.getData().get("htmlUrl");
        String fileName = (String) event.getData().get("fileName");
        String suffix = (String) event.getData().get("suffix");

        //拼接 cmd 命令
        //注意语法格式，空格
        String cmd = wkImageCommand + " --quality 75 " + htmlUrl + " "
                + wkImageStorage + "\\" + fileName + suffix;
        //执行命令 cmd
        try {
            Runtime.getRuntime().exec(cmd);
            logger.info("生成长图成功: " + cmd);
        } catch (IOException e) {
            logger.error("生成长图失败: " + e.getMessage());
        }

        //因为执行 cmd 是异步的，而且输出 生成图片成功 的时候，图片还不一定生成好了
        //我们必须等待图片生成好了，才能进行上传到 七牛云云服务器
        UploadTask uploadTask = new UploadTask(fileName,suffix);
        //每 0.5 秒执行一次，监视图片是否生成成功，一旦生成就上传OSS
        Future future = taskScheduler.scheduleAtFixedRate(uploadTask, 500);

        //因为这句话很快就会被执行了，所以 future 就能有值，不然报错 空指针null
        //执行第一次的定时任务返回的 future
        //不进行设置的话，定时任务停止不了，因为 future.cancel(true)  报错空指针
        //这个玩意就是 定时任务 的线程
        uploadTask.setFuture(future);
    }

    //判断上传的图片的情况，匿名对象
    //服务器直传，上传图片
    class UploadTask implements Runnable {

        // 文件名称
        private String fileName;
        // 文件后缀
        private String suffix;
        // 启动任务的返回值，就是启动的那个定时任务的线程
        private Future future;
        // 开始时间，超过 30 秒判断为生成图片失败
        private long startTime;
        // 上传次数，上传超过 3 次判断为服务器或者网络出现了问题
        private int uploadTimes;

        public UploadTask(String fileName, String suffix) {
            this.fileName = fileName;
            this.suffix = suffix;
            this.startTime = System.currentTimeMillis();
        }

        public void setFuture(Future future) {
            this.future = future;
        }

        @Override
        public void run() {
            // 生成失败，超过 30 秒
            if (System.currentTimeMillis() - startTime > 30000) {
                logger.error("执行时间过长,终止任务:" + fileName);
                //true，终止定时任务执行
                future.cancel(true);
                return;
            }
            // 上传失败，上传超过 3 次
            if (uploadTimes >= 3) {
                logger.error("上传次数过多,终止任务:" + fileName);
                //true，终止定时任务执行
                future.cancel(true);
                return;
            }

            //这个是 生成到本地计算机的 图片
            String path = wkImageStorage + "/" + fileName + suffix;
            File file = new File(path);

            //图片生成好之后，执行上传
            //因为我们需要进行等待 30s 来判断图片是否生成，0.5秒 判断一次
            if (file.exists()) {
                logger.info(String.format("开始第%d次上传[%s].", ++uploadTimes, fileName));
                // 设置响应信息
                StringMap policy = new StringMap();
                policy.put("returnBody", GenerateUtil.getJsonString(0));
                // 生成上传凭证
                Auth auth = Auth.create(accessKey, secretKey);
                String uploadToken = auth.uploadToken(shareBucketName, fileName, 3600, policy);
                // 指定上传机房（存储区域）,华南地区
                UploadManager manager = new UploadManager(new Configuration(Zone.zone2()));
                try {
                    // 开始上传图片，服务器直传
                    Response response = manager.put(
                            path, fileName, uploadToken, null, "image/" + suffix, false);
                    // 处理响应结果
                    JSONObject json = JSONObject.parseObject(response.bodyString());
                    if (json == null || json.get("code") == null || !json.get("code").toString().equals("0")) {
                        logger.info(String.format("第%d次上传失败[%s].", uploadTimes, fileName));
                    } else {
                        logger.info(String.format("第%d次上传成功[%s].", uploadTimes, fileName));
                        //true，终止定时任务执行
                        future.cancel(true);
                    }
                } catch (QiniuException e) {
                    logger.info(String.format("第%d次上传失败[%s].", uploadTimes, fileName));
                }
            } else {
                logger.info("等待图片生成[" + fileName + "].");
            }
        }
    }
}

